from selenium.webdriver.common.by import By

from common.base import Base


class OrderSuccessPage(Base):
    def order_success(self):
        result = self.find_element_text((By.XPATH, '/html/body/div[6]/div/h6'))
        prompt, order_number = result.split(':')
        return prompt
